package com.cleverest.cleverestdiary.service.impl;

import com.cleverest.cleverestdiary.service.FileService;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;

@Service
public class FileServiceImpl implements FileService {

    @Override
    public String getFileMD5(MultipartFile file) throws IOException, NoSuchAlgorithmException {
        MessageDigest md = MessageDigest.getInstance("MD5");

        byte[] fileBytes = file.getBytes();  // 获取文件的字节数组
        md.update(fileBytes);  // 更新 MessageDigest

        byte[] digest = md.digest();  // 获取 MD5 哈希值
        StringBuilder sb = new StringBuilder();
        for (byte b : digest) {
            sb.append(String.format("%02x", b));  // 格式化为 16 进制
        }
        return sb.toString();
    }

    @Override
    public String getFileExtension(MultipartFile file) {
        String fileName = file.getOriginalFilename();
        int dotIndex = fileName.lastIndexOf('.');
        if (dotIndex > 0) {
            return fileName.substring(dotIndex);  // 返回文件扩展名
        } else {
            return "";  // 没有扩展名
        }
    }

    @Override
    public String renameFile(MultipartFile file) throws IOException, NoSuchAlgorithmException {
        // 计算文件的 MD5 值
        String md5Value = getFileMD5(file);

        // 获取文件扩展名
        String fileExtension = getFileExtension(file);

        // 生成新文件名
        String newFileName = md5Value + fileExtension;
        return newFileName;
    }
}
